package com.raven.craft.code.generator.mysql.schema.puml;

import com.raven.craft.code.generator.config.MysqlContext;
import com.raven.craft.code.generator.mysql.repo.entity.Table;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

import java.util.List;

/**
 * @author raven
 * @date 2019-06-06 12:04
 */
@Getter
@Setter
@ToString
public class PumlSchemaVo {

    private Boolean withType;
    private Boolean withComment;
    private List<Table> tables;

    public PumlSchemaVo(MysqlContext context, PumlSchemaProperties properties) {
        this.withType = properties.getWithType();
        this.withComment = properties.getWithComment();
        this.tables = context.getTables();
    }

}
